# coding=utf-8
from django.contrib import admin

from book.models import Book, Hero

# Register your models here.
class BookAdmin(admin.ModelAdmin):
    list_display = ["id", "title"]

class HeroAdmin(admin.ModelAdmin):
    list_display = ["id", "name", "book"]

admin.site.register(Book, BookAdmin)
admin.site.register(Hero, HeroAdmin)